

import torch
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np
import os

print(torch.cuda.device_count())
print(torch.cuda.get_device_name(1))
print(torch.cuda.current_device())

infos = [torch.cuda.get_device_properties(i) for i in range(4)]
print(infos)
#
# path = "/mnt/ssd/dk-projects/multifeature/test/object/bottle/bottle_broken_large_001.png"
# lena = mpimg.imread(path)
# plt.imshow(lena) # 显示图片



print(os.getcwd() )#获取当前工作目录路径

print(os.path.abspath('.') )#获取当前工作目录路径

